package com.tduck.cloud.api.config;

import com.baomidou.mybatisplus.annotation.DbType;
import com.baomidou.mybatisplus.extension.plugins.MybatisPlusInterceptor;
import com.baomidou.mybatisplus.extension.plugins.inner.DataPermissionInterceptor;
import com.baomidou.mybatisplus.extension.plugins.inner.PaginationInnerInterceptor;
import org.apache.ibatis.mapping.DatabaseIdProvider;
import org.apache.ibatis.mapping.VendorDatabaseIdProvider;
import org.mybatis.spring.annotation.MapperScan;
import org.springframework.context.annotation.Bean;
import org.springframework.context.annotation.Configuration;
import org.springframework.transaction.annotation.EnableTransactionManagement;

import java.util.Properties;

/**
 * @author smalljop
 * @description mybaits plus 配置文件
 * @create 2019-12-10 16:28
 **/
@EnableTransactionManagement
@Configuration
@MapperScan("com.tduck.cloud.**.mapper")
public class MybatisPlusConfig {

    /**
     * 新的分页插件,一缓和二缓遵循mybatis的规则,需要设置
     * MybatisConfiguration#useDeprecatedExecutor = false 避免缓存出现问题(该属性会在旧插件移除后一同移除)
     */
    @Bean
    public MybatisPlusInterceptor mybatisPlusInterceptor() {
        // 添加数据权限插件
//        DataPermissionInterceptor dataPermissionInterceptor = new DataPermissionInterceptor();
//        AutoDataPermissionHandler autoDataPermissionHandler = new AutoDataPermissionHandler();
//        dataPermissionInterceptor.setDataPermissionHandler(autoDataPermissionHandler);
        MybatisPlusInterceptor interceptor = new MybatisPlusInterceptor();
//        interceptor.addInnerInterceptor(dataPermissionInterceptor);
        interceptor.addInnerInterceptor(new PaginationInnerInterceptor(DbType.MYSQL));

        return interceptor;
    }


    /**
     * mybatis多数据支持配置
     *
     * @return
     */
    @Bean
    public DatabaseIdProvider databaseIdProvider() {
        DatabaseIdProvider databaseIdProvider = new VendorDatabaseIdProvider();
        Properties p = new Properties();
        p.setProperty("Oracle", "oracle");
        p.setProperty("MySQL", "mysql");
        p.setProperty("PostgreSQL", "pgsql");
        p.setProperty("DB2", "db2");
        p.setProperty("SQL Server", "sqlserver");
        p.setProperty("DM DBMS", "dm");
        databaseIdProvider.setProperties(p);
        return databaseIdProvider;
    }


}
